package com.hippy.erya.config.interceptor;

import com.hippy.erya.config.threadLocal.ActionContext;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;


@Slf4j
@Component
public class ReqValidityInterceptor implements HandlerInterceptor {


    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        if (handler instanceof HandlerMethod) {

            Map<String,String> requestMap = new HashMap<>();
            //获取所有请求头和值
            Enumeration enumeration = request.getHeaderNames();
            while (enumeration.hasMoreElements()) {
                String name = (String) enumeration.nextElement();
                String value = request.getHeader(name);

                requestMap.put(name,value);
            }

            Map<String, Object> mapContext = ActionContext.getMapContext();
            if (mapContext == null) {
                mapContext = new HashMap<>(4);
            }
            mapContext.put("requestMap", requestMap);
            ActionContext.setMapContext(mapContext);

            return true;
        }

        return true;
    }


    @Override
    public void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {
        // 清理ThredLocal当前线程的资源
        ActionContext.remove();
    }


    @Override
    public void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {

    }


}
